package com.baidu.springcloud.util;

import cn.hutool.core.lang.Snowflake;
import cn.hutool.core.net.NetUtil;
import cn.hutool.core.util.IdUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

/**
 * @author ShiLei
 * @version v1.0
 * @date 2020-04-15 20:49
 *
 * 雪花算法生成全局唯一ID 工具类
 */
@Slf4j
@Component
public class IdGeneratorSnowflake {

    /**
     *
     */
    private long workerId = 0;
    /**
     *
     */
    private long datacenterId = 1;

    /**
     * 糊涂工具包依赖下创建雪花算法
     */
    private Snowflake snowflake = IdUtil.createSnowflake(workerId,datacenterId);

    @PostConstruct
    public void init (){
        try {

            workerId = NetUtil.ipv4ToLong(NetUtil.getLocalhostStr());
            log.info("当前机器的workerId:{}",workerId);
        }catch (Exception e){
            e.printStackTrace();
            log.warn("当前机器的workerId获取失败",e);
            workerId = NetUtil.getLocalhostStr().hashCode();
        }
    }

    /**
     * 如果使用默认的 workerId 与 datacenterId
     * 获取下一个ID 返回
     * @return 返回新的ID
     */
    public synchronized long snowflakeId(){
        return snowflake.nextId();
    }

    /**
     *可以自定义 workerId 与 datacenterId 的数量来获取下一个ID
     * workerId  和 datacenterId 的范围在 1-31
     * @param workerId
     * @param datacenterId 数据中心 id
     * @return 返回id
     */
    public synchronized long snowflakeId(long workerId, long datacenterId){
        return snowflake.nextId();
    }

    public static void main(String[] args) {
        System.out.println(new Snowflake(0,1).nextId());
    }
}
